# -*- coding: utf-8 *-*

#   This file is part of Sql-Rice.
#
#    Sql-Rice is free software; you can redistribute it and/or modify
#    it under the terms of the GNU General Public License as published by
#    the Free Software Foundation; either version 3 of the License, or
#    (at your option) any later version.
#
#    Sql-Rice is distributed in the hope that it will be useful,
#    but WITHOUT ANY WARRANTY; without even the implied warranty of
#    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
#    GNU General Public License for more details.
#
#    You should have received a copy of the GNU General Public License
#    along with Sql-Rice; if not, write to the Free Software
#    Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301 USA

import os
import sqlite3
import tempfile


class TempDb(object):

    def __init__(self, script):
        self.path = ''
        self.script = script

    def __enter__(self):
        self.path = tempfile.mktemp()
        conn = sqlite3.connect(self.path)
        conn.executescript(self.script)
        conn.commit()
        conn.close()
        return self

    def __exit__(self, type, value, traceback):
        os.remove(self.path)
